package tips.p_1000.p351_400;

import java.util.Map;
import java.util.TreeMap;

/**
 * 给你一个由非负整数 a1, a2, ..., an 组成的数据流输入，
 * 请你将到目前为止看到的数字总结为不相交的区间列表。
 * 实现 SummaryRanges 类：
 * SummaryRanges() 使用一个空数据流初始化对象。
 * void addNum(int val) 向数据流中加入整数 val 。
 * int[][] getIntervals() 以不相交区间 [starti, endi] 的列表形式返回对数据流中整数的总结。
 * <p>示例：
 * 输入：
 * ["SummaryRanges", "addNum", "getIntervals", "addNum", "getIntervals", "addNum", "getIntervals", "addNum", "getIntervals", "addNum", "getIntervals"]
 * [[], [1], [], [3], [], [7], [], [2], [], [6], []]
 * 输出：
 * [null, null, [[1, 1]], null, [[1, 1], [3, 3]], null, [[1, 1], [3, 3], [7, 7]], null, [[1, 3], [7, 7]], null, [[1, 3], [6, 7]]]
 * <p>
 * 解释：
 * SummaryRanges summaryRanges = new SummaryRanges();
 * summaryRanges.addNum(1);      // arr = [1]
 * summaryRanges.getIntervals(); // 返回 [[1, 1]]
 * summaryRanges.addNum(3);      // arr = [1, 3]
 * summaryRanges.getIntervals(); // 返回 [[1, 1], [3, 3]]
 * summaryRanges.addNum(7);      // arr = [1, 3, 7]
 * summaryRanges.getIntervals(); // 返回 [[1, 1], [3, 3], [7, 7]]
 * summaryRanges.addNum(2);      // arr = [1, 2, 3, 7]
 * summaryRanges.getIntervals(); // 返回 [[1, 3], [7, 7]]
 * summaryRanges.addNum(6);      // arr = [1, 2, 3, 6, 7]
 * summaryRanges.getIntervals(); // 返回 [[1, 3], [6, 7]]
 * <p>提示：
 * 0 <= val <= 10^4
 * 最多调用 addNum 和 getIntervals 方法 3 * 10^4 次
 * <p>进阶：如果存在大量合并，并且与数据流的大小相比，不相交区间的数量很小，该怎么办?
 *
 * @author hc
 */
public class Demo352 {

    class SummaryRanges {
        TreeMap<Integer, Integer> treeMap;

        public SummaryRanges() {
            treeMap = new TreeMap<>();
        }

        public void addNum(int val) {
            Integer l = treeMap.floorKey(val);
            if (l != null && l <= val && treeMap.get(l) >= val) {
                return;
            }
            Integer r = treeMap.ceilingKey(val);
            if (r != null && r == val + 1) {
                treeMap.put(val, treeMap.get(r));
                treeMap.remove(r);
            } else {
                treeMap.put(val, val);
            }
            if (l != null && treeMap.get(l) == val - 1) {
                treeMap.put(l, treeMap.get(val));
                treeMap.remove(val);
            }
        }

        public int[][] getIntervals() {
            int[][] ans = new int[treeMap.size()][2];
            int idx = 0;
            for (Map.Entry<Integer, Integer> entry : treeMap.entrySet()) {
                ans[idx][0] = entry.getKey();
                ans[idx++][1] = entry.getValue();
            }
            return ans;
        }
    }
}
